import pandas as pd
import itertools

# 读取原始Excel文件
df = pd.read_excel('/Users/Cench/Desktop/VTxSTxTQ_1.xlsx')

# 获取VT、ST和TQ的所有可能组合
vt_categories = [1, 2, 3]
st_categories = [1, 2]
tq_categories = [1, 2]
combinations = list(itertools.product(vt_categories, st_categories, tq_categories))

# 创建新的DataFrame用于存储结果
result_df = pd.DataFrame(columns=['VTxSTxTQ', 'Transmission_Ln'])

# 遍历每个组合，并获取相应的Transmission_Ln值
for combination in combinations:
    vt, st, tq = combination
    combination_name = f'VT{vt}xST{st}xTQ{tq}'
    filtered_data = df[(df['VT'] == vt) & (df['ST'] == st) & (df['TQ'] == tq)]
    transmission_ln_values = filtered_data['Transmission_Ln'].tolist()
    for transmission_ln in transmission_ln_values:
        result_df = result_df.append({'VTxSTxTQ': combination_name, 'Transmission_Ln': transmission_ln}, ignore_index=True)

# 保存结果到新的Excel文件
result_df.to_excel('/Users/Cench/Desktop/VTxSTxTQ_2.xlsx', index=False)
